Skip to content

[docs] Add NeMo Automodel training guide#13306

Merged
stevhliu merged 20 commits intohuggingface:mainfrom
pthombre:automodel_docs
Mar 30, 2026
Merged

[docs] Add NeMo Automodel training guide#13306
stevhliu merged 20 commits intohuggingface:mainfrom
pthombre:automodel_docs

Conversation

@pthombre
Copy link
Copy Markdown
Contributor


What does this PR do?

Adds a new documentation page for NeMo Automodel, NVIDIA's PyTorch DTensor-native training library for fine-tuning and pretraining
diffusion models at scale. NeMo Automodel integrates directly with Diffusers — it loads pretrained models from the Hugging Face Hub using
Diffusers model classes and generates outputs via Diffusers pipelines with no checkpoint conversion needed.

The new guide covers:

  • Supported models (Wan 2.1, FLUX.1-dev, HunyuanVideo 1.5)
  • Installation
  • Data preparation and preprocessing
  • Training configuration (annotated YAML reference)
  • Single-node and multi-node training launch
  • Generation / inference with fine-tuned checkpoints
  • How NeMo Automodel integrates with the Diffusers ecosystem
  • Hardware requirements

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you read the contributor guideline?
  • Did you read our philosophy doc (important for complex PRs)?
  • Was this discussed/approved via a GitHub issue or the forum? Please add a link to it if that's the case.
  • Did you make sure to update the documentation with your changes? Here are the
    documentation guidelines, and
    here are tips on formatting docstrings.
  • Did you write any new necessary tests?

Who can review?

@stevhliu @sayakpaul

Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@sayakpaul sayakpaul requested a review from stevhliu March 23, 2026 03:22
Copy link
Copy Markdown
Member

@stevhliu stevhliu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

super nice, thanks for the docs!

Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
pthombre and others added 16 commits March 23, 2026 18:24
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
adding contacts into the readme
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
@pthombre pthombre marked this pull request as ready for review March 27, 2026 23:52
Signed-off-by: Pranav Prashant Thombre <pthombre@nvidia.com>
@pthombre pthombre requested a review from stevhliu March 28, 2026 00:06
@pthombre
Copy link
Copy Markdown
Contributor Author

Thanks a lot for your comments @stevhliu @sayakpaul . The issues should be addressed now!

Copy link
Copy Markdown
Member

@stevhliu stevhliu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks, just a few more minor nits. otherwise lgtm and we can merge!

Comment thread docs/source/en/training/nemo_automodel.md Outdated
Comment thread docs/source/en/training/nemo_automodel.md Outdated
pthombre and others added 2 commits March 30, 2026 09:59
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@stevhliu stevhliu merged commit 7e463ea into huggingface:main Mar 30, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants